Node / meshtastic / Meshtastic-Android / files / feature / docs / src / commonTest / kotlin / org / meshtastic / feature / docs / ai / ChirpySessionHolderTest.kt

Displaying Raw • Download

T8b949e/*

* Copyright (c) 2026 Meshtastic LLC

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <https://www.gnu.org/licenses/>.

*/

Tff7b72package T7ee787org.meshtastic.feature.docs.ai

Tff7b72import T7ee787kotlinx.coroutines.flow.Flow

Tff7b72import T7ee787kotlinx.coroutines.flow.MutableStateFlow

Tff7b72import T7ee787kotlinx.coroutines.flow.StateFlow

Tff7b72import T7ee787kotlinx.coroutines.flow.asFlow

Tff7b72import T7ee787kotlinx.coroutines.flow.emptyFlow

Tff7b72import T7ee787kotlinx.coroutines.flow.flow

Tff7b72import T7ee787kotlinx.coroutines.test.StandardTestDispatcher

Tff7b72import T7ee787kotlinx.coroutines.test.advanceUntilIdle

Tff7b72import T7ee787kotlinx.coroutines.test.runTest

Tff7b72import T7ee787org.meshtastic.core.di.CoroutineDispatchers

Tff7b72import T7ee787org.meshtastic.feature.docs.model.AIDocAssistantResult

Tff7b72import T7ee787org.meshtastic.feature.docs.model.ChirpyRole

Tff7b72import T7ee787org.meshtastic.feature.docs.model.ModelReadiness

Tff7b72import T7ee787kotlin.test.Test

Tff7b72import T7ee787kotlin.test.assertEquals

Tff7b72import T7ee787kotlin.test.assertFalse

Tff7b72import T7ee787kotlin.test.assertTrue

T8b949e/**

* Requests are owned by the holder, not by the pane that submitted them: the docs layout disposes the list pane as soon

* as a page opens, so a request launched from a `rememberCoroutineScope()` was cancelled mid-answer and stranded

* `isLoading = true` in this app-wide singleton, wedging the composer for the rest of the process.

*/

Tff7b72class T56d364ChirpySessionHolderTest Tb4b4b4{

Tf0883e@Test

Tff7b72fun Td2a8ff`submit streams the answer into the shared session`Tb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3results Tff7b72= Te6edf3listOfTb4b4b4(Te6edf3partialTb4b4b4(Ta5d6ff"Ta5d6ffMeshTa5d6ff"Tb4b4b4)Tb4b4b4, Te6edf3successTb4b4b4(Ta5d6ff"Ta5d6ffMeshtastic uses LoRa.Ta5d6ff"Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3holder Tff7b72= Te6edf3holderForTb4b4b4(Te6edf3FakeDocAssistantTb4b4b4(Te6edf3stream Tff7b72= Te6edf3resultsTb4b4b4.Te6edf3asFlowTb4b4b4(Tb4b4b4)Tb4b4b4)Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3sessionState Tff7b72= Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3copyTb4b4b4(Te6edf3draftQuestion Tff7b72= Ta5d6ff"Ta5d6ffWhat is Meshtastic?Ta5d6ff"Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3advanceUntilIdleTb4b4b4(Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(Te6edf3listOfTb4b4b4(Te6edf3ChirpyRoleTb4b4b4.Te6edf3USERTb4b4b4, Te6edf3ChirpyRoleTb4b4b4.Te6edf3ASSISTANTTb4b4b4)Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3messagesTb4b4b4.Te6edf3map Tb4b4b4{ Tffa657itTb4b4b4.Te6edf3role Tb4b4b4}Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(Ta5d6ff"Ta5d6ffMeshtastic uses LoRa.Ta5d6ff"Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3messagesTb4b4b4.Te6edf3lastTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3textTb4b4b4)

Te6edf3assertEqualsTb4b4b4(Ta5d6ff"Ta5d6ff"Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3draftQuestionTb4b4b4)

Te6edf3assertFalseTb4b4b4(Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3isLoadingTb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ff`submit clears loading when the stream ends without a terminal result`Tb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3holder Tff7b72= Te6edf3holderForTb4b4b4(Te6edf3FakeDocAssistantTb4b4b4(Te6edf3stream Tff7b72= Te6edf3emptyFlowTb4b4b4(Tb4b4b4)Tb4b4b4)Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3sessionState Tff7b72= Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3copyTb4b4b4(Te6edf3draftQuestion Tff7b72= Ta5d6ff"Ta5d6ffAnyone there?Ta5d6ff"Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3advanceUntilIdleTb4b4b4(Tb4b4b4)

Te6edf3assertFalseTb4b4b4(Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3isLoadingTb4b4b4, Ta5d6ff"Ta5d6ffa stream with no terminal result must not wedge the composerTa5d6ff"Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ff`a thrown failure clears loading instead of wedging the composer`Tb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3holder Tff7b72= Te6edf3holderForTb4b4b4(Te6edf3FakeDocAssistantTb4b4b4(Te6edf3stream Tff7b72= Te6edf3flow Tb4b4b4{ Tff7b72throw Te6edf3IllegalStateExceptionTb4b4b4(Ta5d6ff"Ta5d6ffassistant blew upTa5d6ff"Tb4b4b4) Tb4b4b4}Tb4b4b4)Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3sessionState Tff7b72= Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3copyTb4b4b4(Te6edf3draftQuestion Tff7b72= Ta5d6ff"Ta5d6ffWhy?Ta5d6ff"Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3advanceUntilIdleTb4b4b4(Tb4b4b4)

Te6edf3assertFalseTb4b4b4(Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3isLoadingTb4b4b4, Ta5d6ff"Ta5d6ffa failed request must not block every later questionTa5d6ff"Tb4b4b4)

T8b949e// The composer is usable again: the next question is accepted rather than dropped by the isLoading guard.

Te6edf3holderTb4b4b4.Te6edf3sessionState Tff7b72= Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3copyTb4b4b4(Te6edf3draftQuestion Tff7b72= Ta5d6ff"Ta5d6ffAgain?Ta5d6ff"Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff2Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3messagesTb4b4b4.Te6edf3count Tb4b4b4{ Tffa657itTb4b4b4.Te6edf3role Tff7b72=Tff7b72= Te6edf3ChirpyRoleTb4b4b4.Te6edf3USER Tb4b4b4}Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ff`submit ignores a blank draft and a request already in flight`Tb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3holder Tff7b72= Te6edf3holderForTb4b4b4(Te6edf3FakeDocAssistantTb4b4b4(Te6edf3stream Tff7b72= Te6edf3listOfTb4b4b4(Te6edf3partialTb4b4b4(Ta5d6ff"Ta5d6ff...Ta5d6ff"Tb4b4b4)Tb4b4b4)Tb4b4b4.Te6edf3asFlowTb4b4b4(Tb4b4b4)Tb4b4b4)Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3assertTrueTb4b4b4(Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3messagesTb4b4b4.Te6edf3isEmptyTb4b4b4(Tb4b4b4)Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3sessionState Tff7b72= Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3copyTb4b4b4(Te6edf3draftQuestion Tff7b72= Ta5d6ff"Ta5d6ffFirst?Ta5d6ff"Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3sessionState Tff7b72= Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3copyTb4b4b4(Te6edf3draftQuestion Tff7b72= Ta5d6ff"Ta5d6ffSecond?Ta5d6ff"Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3submitTb4b4b4(Te6edf3currentPageId Tff7b72= Tff7b72nullTb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3messagesTb4b4b4.Te6edf3count Tb4b4b4{ Tffa657itTb4b4b4.Te6edf3role Tff7b72=Tff7b72= Te6edf3ChirpyRoleTb4b4b4.Te6edf3USER Tb4b4b4}Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(Ta5d6ff"Ta5d6ffSecond?Ta5d6ff"Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3draftQuestionTb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ff`introduce runs once and resets the assistant session`Tb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3assistant Tff7b72= Te6edf3FakeDocAssistantTb4b4b4(Te6edf3stream Tff7b72= Te6edf3emptyFlowTb4b4b4(Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3holder Tff7b72= Te6edf3holderForTb4b4b4(Te6edf3assistantTb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3introduceTb4b4b4(Tb4b4b4)

Te6edf3advanceUntilIdleTb4b4b4(Tb4b4b4)

Te6edf3holderTb4b4b4.Te6edf3introduceTb4b4b4(Tb4b4b4)

Te6edf3advanceUntilIdleTb4b4b4(Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3assistantTb4b4b4.Te6edf3resetCountTb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3messagesTb4b4b4.Te6edf3sizeTb4b4b4)

Te6edf3assertFalseTb4b4b4(Te6edf3holderTb4b4b4.Te6edf3sessionStateTb4b4b4.Te6edf3isLoadingTb4b4b4)

Tb4b4b4}

Tff7b72private Tff7b72fun Te6edf3kotlinxTb4b4b4.Td2a8ffcoroutinesTb4b4b4.Te6edf3testTb4b4b4.Te6edf3TestScopeTb4b4b4.Te6edf3holderForTb4b4b4(Te6edf3assistantTb4b4b4: Te6edf3AIDocAssistantTb4b4b4)Tb4b4b4: Te6edf3ChirpySessionHolder Tb4b4b4{

Tff7b72val Te6edf3dispatcher Tff7b72= Te6edf3StandardTestDispatcherTb4b4b4(Te6edf3testSchedulerTb4b4b4)

Tff7b72return Te6edf3ChirpySessionHolderTb4b4b4(

Te6edf3aiAssistant Tff7b72= Te6edf3assistantTb4b4b4,

Te6edf3dispatchers Tff7b72= Te6edf3CoroutineDispatchersTb4b4b4(Te6edf3io Tff7b72= Te6edf3dispatcherTb4b4b4, Te6edf3main Tff7b72= Te6edf3dispatcherTb4b4b4, Te6edf3default Tff7b72= Te6edf3dispatcherTb4b4b4)Tb4b4b4,

Tb4b4b4)

Tb4b4b4}

Tb4b4b4}

Tff7b72private Tff7b72fun Td2a8ffpartialTb4b4b4(Te6edf3answerTb4b4b4: Tffa657StringTb4b4b4) Tff7b72= Te6edf3AIDocAssistantResultTb4b4b4.Te6edf3PartialTb4b4b4(Te6edf3answerTb4b4b4, Te6edf3emptyListTb4b4b4(Tb4b4b4)Tb4b4b4, Te6edf3usedOnDeviceModel Tff7b72= Tff7b72trueTb4b4b4)

Tff7b72private Tff7b72fun Td2a8ffsuccessTb4b4b4(Te6edf3answerTb4b4b4: Tffa657StringTb4b4b4) Tff7b72= Te6edf3AIDocAssistantResultTb4b4b4.Te6edf3SuccessTb4b4b4(Te6edf3answerTb4b4b4, Te6edf3emptyListTb4b4b4(Tb4b4b4)Tb4b4b4, Te6edf3usedOnDeviceModel Tff7b72= Tff7b72trueTb4b4b4)

Tff7b72private Tff7b72class T56d364FakeDocAssistantTb4b4b4(

Tff7b72private Tff7b72val Te6edf3streamTb4b4b4: Te6edf3FlowTff7b72<Te6edf3AIDocAssistantResultTff7b72>Tb4b4b4,

Tff7b72private Tff7b72val Te6edf3answerTb4b4b4: Te6edf3AIDocAssistantResult Tff7b72= Te6edf3successTb4b4b4(Ta5d6ff"Ta5d6ffHi, I'm Chirpy.Ta5d6ff"Tb4b4b4)Tb4b4b4,

Tb4b4b4) Tb4b4b4: Te6edf3AIDocAssistant Tb4b4b4{

Tff7b72var Te6edf3resetCount Tff7b72= T79c0ff0

Te6edf3private Tff7b72set

Tff7b72override Tff7b72val Te6edf3modelStatusTb4b4b4: Te6edf3StateFlowTff7b72<Te6edf3ModelReadinessTff7b72> Tff7b72= Te6edf3MutableStateFlowTb4b4b4(Te6edf3ModelReadinessTb4b4b4.Te6edf3AvailableTb4b4b4)

Tff7b72override Tff7b72suspend Tff7b72fun Td2a8ffisSupportedTb4b4b4(Tb4b4b4)Tb4b4b4: Tffa657Boolean Tff7b72= Tff7b72true

Tff7b72override Tff7b72suspend Tff7b72fun Td2a8ffanswerTb4b4b4(Te6edf3questionTb4b4b4: Tffa657StringTb4b4b4, Te6edf3currentPageIdTb4b4b4: Tffa657String?Tb4b4b4)Tb4b4b4: Te6edf3AIDocAssistantResult Tff7b72= Te6edf3answer

Tff7b72override Tff7b72fun Td2a8ffanswerStreamTb4b4b4(Te6edf3questionTb4b4b4: Tffa657StringTb4b4b4, Te6edf3currentPageIdTb4b4b4: Tffa657String?Tb4b4b4)Tb4b4b4: Te6edf3FlowTff7b72<Te6edf3AIDocAssistantResultTff7b72> Tff7b72= Te6edf3stream

Tff7b72override Tff7b72fun Td2a8ffresetSessionTb4b4b4(Tb4b4b4) Tb4b4b4{

Te6edf3resetCountTff7b72+Tff7b72+

Tb4b4b4}

Tb4b4b4}